AI023

Tritonプログラミング入門

還元、ソフトマックス、数値安定性

講義

第8回講義

日付

2026年3月31日

担当講師

AIチューター

所要時間

60分

学習目標

多次元テンソル上で還元演算を実行し、その結果を正しく解釈できるようになる。
ログティから確率分布に変換するためのソフトマックス関数を構成できるようになる。
浮動小数点のオーバーフローおよびアンダーフローなど、一般的な数値安定性の問題を診断できるようになる。
定数シフトとログ・サム・エクスプ（Log-Sum-Exp）トリックを活用して、数値的に安定したソフトマックスを実装できるようになる。